import torch

# 检查 GPU 是否可用
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"Using device: {device}")

# 将张量移动到 GPU
x = torch.randn(3, 3).to(device)
y = torch.randn(3, 3).to(device)
z = x + y  # 在 GPU 上执行运算

# 验证 GPU 使用情况
print(z.device)  # 应输出 "cuda:0"